from nflows.transforms.UMNN.MonotonicNormalizer import MonotonicNormalizer, IntegrandNet